04fbbd1f4e2bfc6e930a32be768195e53c4366db,kafka-scheduler/src/test/java/com/mesosphere/dcos/kafka/scheduler/RecoverySchedulerTest.java,RecoverySchedulerTest,testReplaceNonLastBroker,#,80
Before Change
when(stateStore.fetchTasksNeedingRecovery()).thenReturn(Arrays.asList(replaceTaskInfo));
DefaultRecoveryScheduler recoveryScheduler = getTestKafkaRecoveryScheduler();
List<Protos.OfferID> acceptedOfferIds = recoveryScheduler.resourceOffers(
driver,
Arrays.asList(getTestOfferSufficientForNewBroker()),
Optional.empty());
Assert.assertEquals(1, acceptedOfferIds.size());
Assert.assertEquals(KafkaTestUtils.testOfferId, acceptedOfferIds.get(0).getValue());
verify(driver, times(1)).acceptOffers(
After Change
when(stateStore.fetchStatus(any())).thenReturn(Optional.of(statuses.get(0)));
DefaultRecoveryPlanManager recoveryPlanManager = getTestKafkaRecoveryPlanManager();
DefaultPlanScheduler planScheduler = getTestKafkaScheduler();
List<Protos.OfferID> acceptedOfferIds = new ArrayList<>(
planScheduler.resourceOffers(
driver,
Arrays.asList(getTestOfferSufficientForNewBroker()),
recoveryPlanManager.getCandidates(Collections.emptyList()))
);
Assert.assertEquals(1, acceptedOfferIds.size());
Assert.assertEquals(KafkaTestUtils.testOfferId, acceptedOfferIds.get(0).getValue());